package MyLinkedList;

public class MyLinkedList {
    class ListNode {
        public int val;
        public ListNode next;
        public ListNode prev;

        public ListNode(int val) {
            this.val=val;
        }
    }
    public ListNode head;
    public ListNode last;

    public void display() {
        ListNode cur=head;
        while(cur!=null) {
            System.out.print(cur.val+" ");
            cur=cur.next;
        }
        System.out.println();
    }

    public int size() {
        ListNode cur=head;
        int count=0;
        while(cur!=null) {
            count++;
            cur=cur.next;
        }
        return count;
    }

    public boolean contains(int val) {
        ListNode cur=head;
        while(cur!=null) {
            if(cur.val==val) {
                return true;
            }
            cur=cur.next;
        }
        return false;
    }

    public void addFirst(int val) {
        ListNode node =new ListNode(val);
        if(head==null) {
            head=last=node;
            return;
        }
        node.next=head;
        head.prev=node;
        head=node;
    }

    public void addLast(int val) {
        ListNode node =new ListNode(val);
        if(head==null) {
            head=last=node;
            return;
        }
        last.next=node;
        node.prev=last;
        last=node;
    }

    public void addIndex(int index,int val) throws IndexNotLegalException{
        try{
            checkLegalOfIndex(index);
        }catch(IndexNotLegalException e) {
            e.printStackTrace();
        }
        ListNode node =new ListNode(val);
        if(index==0) {
            addFirst(val);
            return;
        }
        if(index==size()) {
            addLast(val);
            return;
        }
        ListNode cur=findIndexSubOne(index);
        node.next=cur;
        cur.prev.next=node;
        node.prev=cur.prev;
        cur.prev=node;
    }

    public ListNode findIndexSubOne(int index) {
        ListNode cur=head;
        for(int i=1;i<=index;i++) {
            cur=cur.next;
        }
        return cur;
    }

    public void checkLegalOfIndex(int index) {
        if(index<0 || index>size()) {
            throw new IndexNotLegalException("index位置不合法");
        }
    }

    public void remove(int val) {
        ListNode cur=head;
        while(cur!=null) {
            if(cur.val==val) {
                if(cur==head) {
                    head=head.next;
                    if(head==null) {
                        last=null;
                        return;
                    }else {
                        head.prev=null;
                    }
                }else {
                    cur.prev.next=cur.next;
                    if(cur.next==null) {
                        last=last.prev;
                    }else {
                        cur.next.prev=cur.prev;
                    }
                }
                return;
            }
            cur=cur.next;
        }
    }

    public void removeAllKey(int val) {
        ListNode cur=head;
        while(cur!=null) {
            if(cur.val==val) {
                if(cur==head) {
                    head=head.next;
                    if(head==null) {
                        last=null;
                        return;
                    }else {
                        head.prev=null;
                    }
                }else {
                    cur.prev.next=cur.next;
                    if(cur.next==null) {
                        last=last.prev;
                    }else {
                        cur.next.prev=cur.prev;
                    }
                }
//                return;
            }
            cur=cur.next;
        }
    }

    public void clear() {
        ListNode cur=head;
        while(cur!=null) {
            ListNode curN=cur.next;
            cur.prev=null;
            cur.next=null;
            cur=curN;
        }
        head=last=null;
    }

}

